package org.example.task.model.dto;

import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 分页结果
 *
 * @param <T> 结果泛型
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class PageDTO<T> {
    /**
     * 总条数
     */
    private Long total;
    /**
     * 总页数
     */
    private Long pages;
    /**
     * 数据集合
     */
    private List<T> list;

    /**
     * 返回空分页结果
     *
     * @param page MybatisPlus的分页结果
     * @param <V>  目标VO类型
     * @param <P>  原始PO类型
     * @return VO的分页对象
     */
    public static <V, P> PageDTO<V> empty(Page<P> page) {
        return new PageDTO<>(page.getTotal(), page.getPages(), Collections.emptyList());
    }

    /**
     * 将MybatisPlus分页结果转为 VO分页结果
     *
     * @param page    MybatisPlus的分页结果
     * @param voClass 目标VO类型的字节码
     * @param <V>     目标VO类型
     * @param <P>     原始PO类型
     * @return VO的分页对象
     */
    public static <V, P> PageDTO<V> of(Page<P> page, Class<V> voClass) {
        // 1.非空校验
        List<P> records = page.getRecords();
        if (records == null || records.isEmpty()) {
            return empty(page);
        }
        // 2.数据转换
        List<V> vos = BeanUtil.copyToList(records, voClass);
        // 3.数据封装返回
        return new PageDTO<>(page.getTotal(), page.getPages(), vos);
    }

    /**
     * 将MybatisPlus分页结果转为 VO分页结果，允许用户自定义PO到VO的转换方式
     *
     * @param page      MybatisPlus的分页结果
     * @param convertor PO到VO的转换函数
     * @param <V>       目标VO类型
     * @param <P>       原始PO类型
     * @return VO的分页对象
     */
    public static <V, P> PageDTO<V> of(Page<P> page, Function<P, V> convertor) {
        List<P> records = page.getRecords();
        if (records == null || records.isEmpty()) {
            return empty(page);
        }
        List<V> vos = records.stream().map(convertor).collect(Collectors.toList());
        return new PageDTO<>(page.getTotal(), page.getPages(), vos);
    }
}
